Filtering and formatting json data python